Here is what I advise you to do to improve your study habits.
1) Identify you have a studying problem (Which you did, so check)
2) Quit any bad habits that may have been distracting but didn't affect your grades in undergrad (i.e. smoking, drinking, etc.)
3) GET A STUDY GROUP (cannot really...